1a1e594abed0d763b49674eb68ca17eff7fe85d2,datumbox-framework-applications/src/main/java/com/datumbox/framework/applications/nlp/TextClassifier.java,TextClassifier,predict,#String#,206
Before Change
* @return
*/
public Record predict(String text) {
TextClassifier.TrainingParameters trainingParameters = knowledgeBase.getTrainingParameters();
Dataframe testDataset = new Dataframe(knowledgeBase.getConf());
testDataset.add(new Record(new AssociativeArray(
AbstractTextExtractor.newInstance(trainingParameters.getTextExtractorClass(), trainingParameters.getTextExtractorParameters())
.extract(StringCleaner.clear(text))), null)
);
After Change
* @return
*/
public Record predict(String text) {
TrainingParameters trainingParameters = (TrainingParameters) knowledgeBase.getTrainingParameters();
Dataframe testDataset = new Dataframe(knowledgeBase.getConf());
testDataset.add(